tensor([[0.2700, 0.2700, 0.2700, 0.2700, 0.2700],
        [0.7300, 0.7300, 0.7300, 0.7300, 0.7300]], grad_fn=<RoundBackward1>)